home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 4 / Apprentice-Release4.iso / Source Code / C / Applications / Python 1.3 / Python 1.3 68K / scripts / EditPythonPrefs.py < prev    next >
Encoding:
Python Source  |  1995-10-11  |  3.8 KB  |  169 lines  |  [TEXT/PYTH]
  1. """Edit the Python Preferences file."""
  2. import addpack
  3. addpack.addpack('Tools')
  4. addpack.addpack('bgen')
  5. addpack.addpack('evt')
  6.  
  7. from Dlg import *
  8. from Events import *
  9. from Res import *
  10. import string
  11. import struct
  12. import macfs
  13. import MacOS
  14. import os
  15. import sys
  16. import Res # For Res.Error
  17.  
  18. # resource IDs in our own resources (dialogs, etc)
  19. MESSAGE_ID = 256
  20.  
  21. DIALOG_ID = 512
  22. TEXT_ITEM = 1
  23. OK_ITEM = 2
  24. CANCEL_ITEM = 3
  25. REVERT_ITEM = 4
  26. DIR_ITEM = 5
  27.  
  28. # Resource IDs in the preferences file
  29. PATH_STRINGS_ID = 128
  30. DIRECTORY_ID = 128
  31.  
  32. WRITE = 2
  33. smAllScripts = -3
  34. kOnSystemDisk = 0x8000
  35.  
  36. def restolist(data):
  37.     """Convert STR# resource data to a list of strings"""
  38.     if not data:
  39.         return []
  40.     num, = struct.unpack('h', data[:2])
  41.     data = data[2:]
  42.     rv = []
  43.     for i in range(num):
  44.         strlen = ord(data[0])
  45.         if strlen < 0: strlen = strlen + 256
  46.         str = data[1:strlen+1]
  47.         data = data[strlen+1:]
  48.         rv.append(str)
  49.     return rv
  50.     
  51. def listtores(list):
  52.     """Convert a list of strings to STR# resource data"""
  53.     rv = struct.pack('h', len(list))
  54.     for str in list:
  55.         rv = rv + chr(len(str)) + str
  56.     return rv
  57.  
  58. def message(str = "Hello, world!", id = MESSAGE_ID):
  59.     """Show a simple alert with a text message"""
  60.     d = GetNewDialog(id, -1)
  61.     print 'd=', d
  62.     tp, h, rect = d.GetDialogItem(2)
  63.     SetDialogItemText(h, str)
  64.     while 1:
  65.         n = ModalDialog(None)
  66.         if n == 1: break
  67.         
  68. def interact(list, pythondir):
  69.     """Let the user interact with the dialog"""
  70.     opythondir = pythondir
  71.     try:
  72.         # Try to go to the "correct" dir for GetDirectory
  73.         os.chdir(pythondir.as_pathname())
  74.     except os.error:
  75.         pass
  76.     d = GetNewDialog(DIALOG_ID, -1)
  77.     tp, h, rect = d.GetDialogItem(1)
  78.     SetDialogItemText(h, string.joinfields(list, '\r'))
  79.     while 1:
  80.         n = ModalDialog(None)
  81.         if n == OK_ITEM:
  82.             break
  83.         if n == CANCEL_ITEM:
  84.             return None
  85.         if n == REVERT_ITEM:
  86.             return [], pythondir
  87.         if n == DIR_ITEM:
  88.             fss, ok = macfs.GetDirectory('Select python home folder:')
  89.             if ok:
  90.                 pythondir = fss
  91.     tmp = string.splitfields(GetDialogItemText(h), '\r')
  92.     rv = []
  93.     for i in tmp:
  94.         if i:
  95.             rv.append(i)
  96.     return rv, pythondir
  97.     
  98. def main():
  99.     try:
  100.         h = OpenResFile('EditPythonPrefs.rsrc')
  101.     except Res.Error:
  102.         pass    # Assume we already have acces to our own resource
  103.     
  104.     # Find the preferences folder and our prefs file, create if needed.    
  105.     vrefnum, dirid = macfs.FindFolder(kOnSystemDisk, 'pref', 0)
  106.     preff_fss = macfs.FSSpec((vrefnum, dirid, 'Python Preferences'))
  107.     try:
  108.         preff_handle = FSpOpenResFile(preff_fss, WRITE)
  109.     except Res.Error:
  110.         # Create it
  111.         message('No preferences file, creating one...')
  112.         FSpCreateResFile(preff_fss, 'PYTH', 'pref', smAllScripts)
  113.         preff_handle = FSpOpenResFile(preff_fss, WRITE)
  114.         
  115.     # Load the path and directory resources
  116.     try:
  117.         sr = GetResource('STR#', PATH_STRINGS_ID)
  118.     except (MacOS.Error, Res.Error):
  119.         message('Cannot find any sys.path resource! (Old python?)')
  120.         sys.exit(0)
  121.     d = sr.data
  122.     l = restolist(d)
  123.     
  124.     try:
  125.         dr = GetResource('alis', DIRECTORY_ID)
  126.         fss, fss_changed = macfs.RawAlias(dr.data).Resolve()
  127.     except (MacOS.Error, Res.Error):
  128.         dr = None
  129.         fss = macfs.FSSpec(os.getcwd())
  130.         fss_changed = 1
  131.     
  132.     # Let the user play away
  133.     result = interact(l, fss)
  134.     
  135.     # See what we have to update, and how
  136.     if result == None:
  137.         sys.exit(0)
  138.         
  139.     pathlist, nfss = result
  140.     if nfss != fss:
  141.         fss_changed = 1
  142.         
  143.     if fss_changed or pathlist != l:
  144.         if fss_changed:
  145.             alias = nfss.NewAlias()
  146.             if dr:
  147.                 dr.data = alias.data
  148.                 dr.ChangedResource()
  149.             else:
  150.                 dr = Resource(alias.data)
  151.                 dr.AddResource('alis', DIRECTORY_ID, '')
  152.                 
  153.         if pathlist != l:
  154.             if pathlist == []:
  155.                 if sr.HomeResFile() == preff_handle:
  156.                     sr.RemoveResource()
  157.             elif sr.HomeResFile() == preff_handle:
  158.                 sr.data = listtores(pathlist)
  159.                 sr.ChangedResource()
  160.             else:
  161.                 sr = Resource(listtores(pathlist))
  162.                 sr.AddResource('STR#', PATH_STRINGS_ID, '')
  163.                 
  164.     CloseResFile(preff_handle)
  165.  
  166. if __name__ == '__main__':
  167.     print # Stupid, to init toolboxes...
  168.     main()
  169.